#!/usr/bin/env python
# coding: utf-8
from utils import general_utils
from paper秭归LSTM import run_many_times

if __name__ == '__main__':
    # 检查设置保存结果的各个文件夹是否存在，不存在则创建
    general_utils.check_data_settings_folder()

    # 不同权重时，模型的精度
    # run_many_times_class_weights(n=10, nb_max_epochs=500, max_weight = 22, node_num=256, nb_mid_layers=2, flatten_neighbours=True)

    # print('----------------------开始权重比值1:1-------------------')
    run_many_times(n=1, nb_max_epochs=2,
                   cls_weight={0: 1, 1: 1},
                   node_num=256, nb_mid_layers=2,
                   save_test_pred_results=False,
                   save_all_pred_results=False,
                   save_fprTpr_to_npy=False,
                   save_trained_model=False,
                   use_early_stop=False,
                   flatten_neighbours=True)     #最后一个参数，是否展开邻居

    # print('----------------------开始权重比值1:4-------------------')
    # run_many_times(n=10, nb_max_epochs=200,
    #                cls_weight={0: 1, 1: 4},
    #                save_test_pred_results=False,
    #                save_all_pred_results=False,
    #                save_fprTpr_to_npy=False,
    #                save_trained_model=False,
    #                use_early_stop = False,
    #                flatten_neighbours=True)

